package com.z.system.service.impl;

import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.z.system.dto.RoleDTO;
import com.z.system.model.Permission;
import com.z.system.model.Role;
import com.z.system.repository.PermissionRepository;
import com.z.system.repository.RoleRepository;
import com.z.system.service.RoleService;

/**
 * 角色服务实现类
 */
@Service
public class RoleServiceImpl implements RoleService {

    @Autowired
    private RoleRepository roleRepository;

    @Autowired
    private PermissionRepository permissionRepository;

    @Override
    public Role getRoleById(UUID id) {
        return roleRepository.findById(id).orElse(null);
    }

    @Override
    public Role getRoleByName(String name) {
        return roleRepository.findByName(name).orElse(null);
    }

    @Override
    public Role getRoleByCode(String code) {
        return roleRepository.findByCode(code).orElse(null);
    }

    @Override
    public Page<Role> getRoles(Pageable pageable, Map<String, Object> params) {
        // 这里可以根据参数构建查询条件
        // 简化实现，直接返回所有角色的分页结果
        return roleRepository.findAll(pageable);
    }

    @Override
    public List<Role> getAllRoles() {
        return roleRepository.findAll();
    }

    @Override
    public Page<RoleDTO> getRoleDTOs(Pageable pageable, Map<String, Object> params) {
        // 从参数中提取搜索条件
        String name = params != null ? (String) params.get("name") : null;
        String code = params != null ? (String) params.get("code") : null;

        // 直接从数据库查询并返回RoleDTO列表，支持搜索参数，避免额外的实体到DTO的转换
        return roleRepository.findRoleDTOsByNameAndCode(pageable, name, code);
    }

    @Override
    public Role saveRole(Role role) {
        // 检查角色名称是否已存在
        if (roleRepository.existsByName(role.getName())) {
            throw new RuntimeException("角色名称已存在");
        }
        // 检查角色标识是否已存在
        if (roleRepository.existsByCode(role.getCode())) {
            throw new RuntimeException("角色标识已存在");
        }
        // 设置创建时间
        if (role.getCreateTime() == null) {
            role.setCreateTime(new Date());
        }
        return roleRepository.save(role);
    }

    @Override
    public Role updateRole(Role role) {
        Role existingRole = roleRepository.findById(role.getId()).orElseThrow(() -> new RuntimeException("角色不存在"));

        // 检查角色名称是否已存在（排除当前角色）
        if (!existingRole.getName().equals(role.getName()) &&
                roleRepository.existsByName(role.getName())) {
            throw new RuntimeException("角色名称已存在");
        }
        // 检查角色标识是否已存在（排除当前角色）
        if (!existingRole.getCode().equals(role.getCode()) &&
                roleRepository.existsByCode(role.getCode())) {
            throw new RuntimeException("角色标识已存在");
        }

        // 设置更新时间
        role.setUpdateTime(new Date());

        return roleRepository.save(role);
    }

    @Override
    public void deleteRole(UUID id) {
        Role role = roleRepository.findById(id).orElseThrow(() -> new RuntimeException("角色不存在"));
        // 检查角色是否被用户使用
        // 简化实现，直接删除
        roleRepository.delete(role);
    }

    @Override
    public void deleteRoles(Set<UUID> ids) {
        List<Role> roles = roleRepository.findAllById(ids);
        if (roles.size() != ids.size()) {
            throw new RuntimeException("部分角色不存在");
        }
        // 检查角色是否被用户使用
        // 简化实现，直接删除
        roleRepository.deleteAll(roles);
    }

    @Override
    @Transactional
    public void assignPermissions(UUID roleId, Set<String> permissionIds) {
        // 检查角色是否存在
        if (!roleRepository.existsById(roleId)) {
            throw new RuntimeException("角色不存在");
        }

        // 将字符串权限ID转换为UUID
        Set<UUID> uuidPermissionIds = new HashSet<>();
        for (String permissionIdStr : permissionIds) {
            try {
                UUID permissionId = UUID.fromString(permissionIdStr);
                uuidPermissionIds.add(permissionId);
            } catch (IllegalArgumentException e) {
                throw new RuntimeException("权限ID格式错误: " + permissionIdStr);
            }
        }

        // 检查权限是否存在
        if (!uuidPermissionIds.isEmpty()) {
            List<Permission> permissions = permissionRepository.findAllById(uuidPermissionIds);
            if (permissions.size() != uuidPermissionIds.size()) {
                throw new RuntimeException("部分权限不存在");
            }
        }

        // 开始事务
        // 删除现有关联
        roleRepository.deleteRolePermissions(roleId);

        // 批量添加新关联
        if (!uuidPermissionIds.isEmpty()) {
            for (UUID permissionId : uuidPermissionIds) {
                roleRepository.addRolePermission(roleId, permissionId);
            }
        }
    }

    @Override
    public Set<UUID> getRolePermissionIds(UUID roleId) {
        // 检查角色是否存在
        if (!roleRepository.existsById(roleId)) {
            throw new RuntimeException("角色不存在");
        }

        // 使用SQL查询获取权限ID集合
        return roleRepository.findPermissionIdsByRoleId(roleId);
    }
}